CropAndResize ================= 从输入图像Tensor中提取切片并调整其大小。仅支持双线性插值方法。 输入: - **src** - 输入数据的地址。 - **box_idx** - boxes的索引,box_idx[i]的值表示第i个框的图像的值。 - **boxes** - 第i行表示box_index[i]图像区域的坐标,并且坐标[y1,x1,y2,x2]是归一化后的值。归一化后的坐标值y,映射到图像y*(image_height-1)处,因此归一化后的图像高度范围为[0,1],映射到实际图像高度范围为[0,image_height-1]。我们允许y1>y2,在这种情况下,视为原始图像的上下翻转变换。宽度尺寸的处理类似。坐标取值允许在[0,1]范围之外,在这种情况下,我们使用extrapolation_value外插值进行补齐。 - **param** - 算子计算所需参数的结构体。其各成员见下述。 - **extrapolation_value** - 外插值。 - **core_mask** - 核掩码。 **CropAndResizeParameter定义:** .. code-block:: c :linenos: typedef struct CropAndResizeParameter { int* input_shape_; // 输入张量形状 int* output_shape_; // 输出张量形状 int* x_lefts_; // 用于存储预处理结果 int* x_rights_; // 用于存储预处理结果 int* y_tops_; // 用于存储预处理结果 int* y_bottoms_; // 用于存储预处理结果 void* x_weights_; // 用于存储预处理结果 void* y_weights_; // 用于存储预处理结果 void* line_buffers_; // 用于存储中间结果 } CropAndResizeParameter; 输出: - **output** - 输出地址。 支持平台: ``FT78NE`` ``MT7004`` .. note:: - FT78NE 支持int8, fp32 - MT7004 支持fp16, fp32 **共享/私有存储版本:** .. c:function:: void i8_crop_and_resize_anycore(int8_t* src, int8_t* dst, int *box_idx, float *boxes, CropAndResizeParameter* param, float extrapolation_value, int core_mask) .. c:function:: void hp_crop_and_resize_anycore(half* src, half* dst, int *box_idx, float *boxes, CropAndResizeParameter* param, float extrapolation_value, int core_mask) .. c:function:: void fp_crop_and_resize_anycore(float* src, float* dst, int *box_idx, float *boxes, CropAndResizeParameter* param, half extrapolation_value, int core_mask) 私有及共享空间版本均使用这些函数。 **C调用示例:** .. code-block:: c :linenos: :emphasize-lines: 28 void TestCropAndResizeSMCFp32(int* input_shape, int* output_shape, float* inp_boxes, int32_t* inp_box_idx, float extrapolation_value, int core_mask) { int core_id = get_core_id(); int core_num = GetCoreNum(core_mask); int logic_core_id = GetLogicCoreId(core_mask, core_id); float* input = (float*)0x88000000; // 测试私有空间时地址设置在私有空间内即可 float* output = (float*)0x89000000; float* boxes = (float*)0x8A000000; int* box_idx = (int*)0x8B000000; CropAndResizeParameter* param = (CropAndResizeParameter*)0x8C000000; if (logic_core_id == 0) { memcpy(boxes, inp_boxes, sizeof(float) * output_shape[0] * 4); memcpy(box_idx, inp_box_idx, sizeof(int) * output_shape[0]); param->input_shape_ = (int*)0x8D000000; memcpy(param->input_shape_, input_shape, sizeof(int) * 4); param->output_shape_ = (int*)0x8E000000; memcpy(param->output_shape_, output_shape, sizeof(int) * 4); param->line_buffers_ = (void*)0x8F000000; param->x_lefts_ = (int*)0x90000000; param->x_rights_ = (int*)0x91000000; param->y_bottoms_ = (int*)0x92000000; param->y_tops_ = (int*)0x93000000; param->x_weights_ = (void*)0x94000000; param->y_weights_ = (void*)0x95000000; PrepareCropAndResizeBilinear(param->input_shape_, boxes, param->output_shape_, param->y_bottoms_, param->y_tops_, param->x_lefts_, param->x_rights_, param->y_weights_, param->x_weights_); // 做预处理 } sys_bar(0, core_num); // 初始化参数完成后进行同步 fp_crop_and_resize_anycore(input, output, box_idx, boxes, param, extrapolation_value, core_mask); } void main(){ int input_shape[4] = {1, 4, 4, 4}; int output_shape[4] = {1, 8, 8, 4}; float boxes[4] = {0, 0, 0.5, 0.5}; int box_idx[1] = {0}; int core_mask = 0b1111; // 测试单核时核掩码设置为0b0001即可 float extrapolation_value = 0.5; TestCropAndResizeSMCFp32(input_shape, output_shape, boxes, box_idx, extrapolation_value, core_mask); }